# 学生管理系统
import os.path
import sys

#  使用全局变量，管理信息
students = []
# 新增学生、显示学生、查找学生、删除、存档到文件

def menu():
    print(' 1. 新增学生')
    print(' 2. 显示学生')
    print(' 3. 查找学生')
    print(' 4. 删除学生')
    print(' 0. 退出程序')
    choice = input('请输入您的选择：')
    return  choice


def insert():
    print('【新增学生】 开始！')
    studentId = input('请输入学生的学号：')
    name = input('请输入学生的姓名：')
    gender = input('请输入学生的性别：')
    if gender not in ('男','女'):
        print('输入性别有误！')
        return
    className = input('请输入班级')
    # 组合信息
    student = {
        'studentId' :studentId,
        'name' :name,
        'gender' :gender,
        'className' :className,
    }
    global students
    students.append(student)
    #     保存
    save()
    print('【新增学生】 结束！')


def show():
    print('【显示学生】 开始！')
    print(f'学号\t姓名\t性别\t班级')
    for s in students:
        print(f"{s['studentId']}\t{s['name']}\t{s['gender']}\t{s['className']}")
    print(f'【显示学生】 结束！共显示了 {len(students)} 条数据')



def find():
    print('【查找学生】 开始！')
    count = 0
    name = input('请输入姓名：')
    print(f'学号\t姓名\t性别\t班级')
    for s in students:
        if name == s['name']:
            print(f"{s['studentId']}\t{s['name']}\t{s['gender']}\t{s['className']}")
            count += 1
    print(f'【查找学生】 结束！共查找了 {count} 个匹配的学生!')

def delete():
    print('【删除学生】 开始！')
    studentId = input('请输入id：')
    for s in students:
        if studentId == s['studentId']:
            print(f"删除{s['name']}的信息")
            students.remove(s)
        else:
            print('未找到！')
        #     保存
        save()
    print('【删除学生】 结束!')


def save():
    """
    用于存档
    """
    # 此处的路径是相对路径
    # #此时这个写法的含义就是让record.txt和当前的code66.py在同一个目录里，
    with open('record.txt','a',encoding='utf8') as f:
        for s in students:
            f.write(f"{s['studentId']}\t{s['name']}\t{s['gender']}\t{s['className']}\n")
        print(f'【存档】成功！共存储了{len(students)} 条记录。')


def load():
    """
    用于读档
    """
    #  如果存档文件不存在，则直接跳过读档流程
    if not os.path.exists('record.txt'):
      return

    # 读档的时候要保证先把旧的数据给清理干净！
    global students
    students = []
    with open('record.txt','r',encoding='utf8') as f:
        for line in f:
        # 针对这一行数据，按照\t进行切分操作！
        # 切分之前，要去除末尾的换行
            # 这是一个非常有用的方法. 功能是去掉一个字符串开头和末尾的空白符~
            line = line.strip()
            tokens = line.split('\t')
            if len(tokens) != 4:
                print(f'当前行格式存在问题！line = {line}')
                continue
            student = {
                'studentId' :tokens[0],
                'name' :tokens[1],
                'gender' :tokens[2],
                'className' :tokens[3]
            }
            students.append(student)
    print(f'【读档成功】，共读取了 {len(students)} 条数据')

def main():
    """
    入口函数
    :return:
    """
    print('-------------------------------')
    print('         欢迎来到学生系统         ')
    print('-------------------------------')
    load()
    while True:
        #通过 menu 函数来打印菜单系统
        choice = menu()
        if choice == '1':
            #新增函数
            insert()
        elif choice == '2':
#
            show()
        elif choice == '3':
#
            find()
        elif choice == '4':
#
            delete()
        elif choice == '0':
            print('goodbye!')
            sys.exit(0)
        else:
            print('您的输入有误，请重新输入！！！')
            

main()